package io.easyspring.common.generator.config;

import com.baomidou.mybatisplus.generator.config.TemplateConfig;

/**
 * 生成器的模板配置类
 *
 * @author summer
 * @version V1.0.0-RELEASE
 * DateTime 2019-05-11 18:52
 */
public class GeneratorTemplateConfig {

    /**
     * 模板配置
     *
     * @return com.baomidou.mybatisplus.generator.config.TemplateConfig
     * Author summer
     * Version V1.0.0-RELEASE
     * DateTime 2019-05-11 18:52
     */
    public static TemplateConfig templateConfig(){
        // 创建模板配置对象
        TemplateConfig templateConfig = new TemplateConfig();

        /*
         * 自定义模板配置，可以 copy 源码 mybatis-plus/src/main/resources/templates 下面内容修改，
         * 放置自己项目的 src/main/resources/templates 目录下, 默认名称一下可以不配置，也可以自定义模板名称
         * 如下任何一个模块如果设置 空 OR Null 将不生成该模块。
         */
        // Java 实体类模板
        templateConfig.setEntity(null);
        // Kotin 实体类模板
        templateConfig.setEntityKt(null);
        // Service 类模板
        templateConfig.setService(null);
        // Service impl 实现类模板
        templateConfig.setServiceImpl(null);
        // controller 控制器模板
        templateConfig.setController(null);
        // mapper 模板
        templateConfig.setMapper(null);
        // mapper xml 模板
        templateConfig.setXml(null);

        return templateConfig;
    }
}
